//
//  UIDevice+BZIdentifier.m
//  BZFoundation
//
//  Created by 郭艺伟 on 2017/9/28.
//  Copyright © 2017年 郭艺伟. All rights reserved.
//

#import "UIDevice+BZIdentifier.h"
#include <sys/sysctl.h>
#include <net/if.h>
#include <net/if_dl.h>
#import <CommonCrypto/CommonDigest.h>

// Return the local MAC addy
// Courtesy of FreeBSD hackers email list
// Accidentally munged during previous update. Fixed thanks to erica sadun & mlamb.
UIKIT_STATIC_INLINE NSString *macaddress() {
    int                 mib[6];
    size_t              len;
    char                *buf;
    unsigned char       *ptr;
    struct if_msghdr    *ifm;
    struct sockaddr_dl  *sdl;
    
    mib[0] = CTL_NET;
    mib[1] = AF_ROUTE;
    mib[2] = 0;
    mib[3] = AF_LINK;
    mib[4] = NET_RT_IFLIST;
    
    if ((mib[5] = if_nametoindex("en0")) == 0) {
        printf("Error: if_nametoindex error\n");
        return NULL;
    }
    
    if (sysctl(mib, 6, NULL, &len, NULL, 0) < 0) {
        printf("Error: sysctl, take 1\n");
        return NULL;
    }
    
    if ((buf = malloc(len)) == NULL) {
        printf("Could not allocate memory. error!\n");
        return NULL;
    }
    
    if (sysctl(mib, 6, buf, &len, NULL, 0) < 0) {
        printf("Error: sysctl, take 2");
        free(buf);
        return NULL;
    }
    
    ifm = (struct if_msghdr *)buf;
    sdl = (struct sockaddr_dl *)(ifm + 1);
    ptr = (unsigned char *)LLADDR(sdl);
    NSString *outstring = [NSString stringWithFormat:@"%02X:%02X:%02X:%02X:%02X:%02X",
                           *ptr, *(ptr+1), *(ptr+2), *(ptr+3), *(ptr+4), *(ptr+5)];
    free(buf);
    
    return outstring;
}

UIKIT_STATIC_INLINE NSString *MD5(NSString *string, bool isShort) {
    if (!string || [string length] == 0) {
        return nil;
    }
    const char *cStr = [string UTF8String];
    unsigned char digest[CC_MD5_DIGEST_LENGTH];
    CC_MD5(cStr, (CC_LONG)strlen(cStr), digest);
    
    NSMutableString *output = nil;
    if (isShort) {
        output = [NSMutableString stringWithFormat:
                  @"%02x%02x%02x%02x%02x%02x%02x%02x",
                  digest[4], digest[5], digest[6], digest[7],
                  digest[8], digest[9], digest[10], digest[11]
                  ];
        
    } else {
        output = [NSMutableString stringWithCapacity:CC_MD5_DIGEST_LENGTH * 2];
        
        for(int i = 0; i < CC_MD5_DIGEST_LENGTH; i++) {
            [output appendFormat:@"%02x", digest[i]];
        }
    }
    
    return output;
}



@implementation UIDevice (BZIdentifier)
- (NSString *)bz_uniqueDeviceIdentifier {
    NSString *bundleIdentifier = [[NSBundle mainBundle] bundleIdentifier];
    NSString *strIdentifier = nil;
    
    // 在iOS6以上，获取identifierForVendor
    if ([[UIDevice currentDevice] respondsToSelector:@selector(identifierForVendor)]) {
        strIdentifier = [[[UIDevice currentDevice] identifierForVendor] UUIDString];
    } else {
        strIdentifier = macaddress();
    }
    
    NSString *stringToHash = [NSString stringWithFormat:@"%@%@", strIdentifier, bundleIdentifier];
    
    return MD5(stringToHash, false);
}

- (NSString *)bz_uniqueGlobalDeviceIdentifier {
    NSString *strIdentifier = nil;
    
    // 在iOS6以上，获取identifierForVendor
    if ([[UIDevice currentDevice] respondsToSelector:@selector(identifierForVendor)]) {
        strIdentifier =[[[UIDevice currentDevice] identifierForVendor] UUIDString];
    } else {
        strIdentifier = macaddress();
    }
    
    return MD5(strIdentifier, false);;
}

@end
